package cn.tedu.ivos.base.file;

import cn.tedu.ivos.base.response.JsonResult;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.UUID;

@Slf4j
@RestController
@RequestMapping("/v1/file")
public class UploadController {

    /**
     * 定义图片上传的方法
     * MultipartFile 用来接收前端上传的图片文件对象
     * @param file 注意:此名字必须与upload上传组件的name名一致!
     */
    @PostMapping("upload")
    public JsonResult upload(MultipartFile file) throws IOException {
        //1.获取原始文件名
        String fileName = file.getOriginalFilename();
        //2.获取图片后缀名
        String suffix = fileName.substring(fileName.lastIndexOf("."));
        //3.生成一个随机串作为新的图片名
        fileName = UUID.randomUUID()+suffix;
        //4.指定一个磁盘路径,作为保存图片文件的根路径
        String dirPath = "d:/files";
        //5.为了提高检索性能,我们可以设置一个日期格式生成对应的文件夹来存储图片文件
        //这里采用的是年月日的格式
        SimpleDateFormat sdf = new SimpleDateFormat("/yyyy/MM/dd/");
        //6.将当前真实的日期转为上面的格式
        String datePath = sdf.format(new Date());
        //7.将上面的磁盘路径与日期路径拼接成一个完整的图片文件存放路径
        File dirFile = new File(dirPath+datePath);
        //8.如果该路径下对应的文件夹不存在,则创建文件夹
        if(!dirFile.exists()){
            dirFile.mkdirs();
        }
        //9.拼接一个完整的图片路径
        String filePath = dirPath+datePath+fileName;
        //10.将图片文件保存到上面的路径下
        file.transferTo(new File(filePath));
        return JsonResult.ok(datePath+fileName);
    }
    /**
     * 定义处理删除图片的方法
     * @param imgUrl 注意:这个路径是日期路径+图片名,不包含磁盘路径!
     */
    @PostMapping("remove")
    public JsonResult remove(String imgUrl){
        log.debug("图片删除:imgUrl="+imgUrl);
        new File("d:/files"+imgUrl).delete();
        return JsonResult.ok();
    }
}
